package com.itheima.service.impl;

import com.alibaba.dubbo.config.annotation.Service;
import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import com.itheima.dao.MenuDao;
import com.itheima.dao.PermissionDao;
import com.itheima.dao.RoleDao;
import com.itheima.dao.UserDao;
import com.itheima.entity.PageResult;
import com.itheima.entity.QueryPageBean;
import com.itheima.pojo.Menu;
import com.itheima.pojo.Permission;
import com.itheima.pojo.Role;
import com.itheima.pojo.User;
import com.itheima.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;

/**
 * 用户服务
 */
@Service(interfaceClass = UserService.class)
@Transactional
public class UserServiceImpl implements UserService {
    @Autowired
    private UserDao userDao;
    @Autowired
    private RoleDao roleDao;
    @Autowired
    private PermissionDao permissionDao;
    @Autowired
    private MenuDao menuDao;
    //根据用户名查询数据库获取用户信息和关联的角色信息，同时需要查询角色关联的权限信息
    @Override
    public User findByUsername(String username) {
        User user = userDao.findByUsername(username);//查询用户基本信息，不包含用户的角色
        if(user == null){
            return null;
        }
        Integer userId = user.getId();
        //根据用户ID查询对应的角色
        Set<Role> roles = roleDao.findByUserId(userId);
        for (Role role : roles) {
            Integer roleId = role.getId();
            //根据角色ID查询关联的权限
            Set<Permission> permissions = permissionDao.findByRoleId(roleId);
            role.setPermissions(permissions);//让角色关联权限
            LinkedHashSet<Menu> menuLinkedHashSet= menuDao.findMenuByRoleId(roleId);
            LinkedHashSet<Menu> menus = new LinkedHashSet<>();
            List<Menu> children;
            for (Menu menu : menuLinkedHashSet) {
                if (menu.getParentMenuId()==null){
                    //是父菜单
                    children= menuDao.findByParentId(menu.getId());
                    menu.setChildren(children);
                    menus.add(menu);
                }
            }
            role.setMenus(menus);//让角色关联菜单
            //判断是否是子菜单
        }
        user.setRoles(roles);//让用户关联角色
        return user;
    }

    @Override
    public void add(User user, Integer[] roleIds) {
        //从user中获取password
        String password = user.getPassword();
        if(password!=null && !"".equals(password)){
            BCryptPasswordEncoder bCryptPasswordEncoder = new BCryptPasswordEncoder();
            password = bCryptPasswordEncoder.encode(password);
            user.setPassword(password);
        }
        //向user表中添加数据
        userDao.add(user);
        //向中间表中添加数据
        if(roleIds!=null && roleIds.length>0){
            for (Integer roleId : roleIds) {
                Map<String,Integer> map = new HashMap<>();
                map.put("userId",user.getId());
                map.put("roleId",roleId);
                userDao.setUserAndRole(map);
            }
        }
    }

    @Override
    public PageResult pageQuery(QueryPageBean queryPageBean) {
        Integer currentPage = queryPageBean.getCurrentPage();
        Integer pageSize = queryPageBean.getPageSize();
        String queryString = queryPageBean.getQueryString();
        PageHelper.startPage(currentPage,pageSize);
        Page<User> page = userDao.findByCondition(queryString);
        return new PageResult(page.getTotal(),page.getResult());
    }

    @Override
    public User findById(Integer id) {
        User user = userDao.findById(id);
        return user;
    }

    @Override
    public List<Integer> findRoleIdsByUserId(Integer id) {
        List<Integer> roleIds = userDao.findRoleIdsByUserId(id);
        return roleIds;
    }

    @Override
    public void edit(User user, Integer[] roleIds) {
        String password = user.getPassword();
        if(password!=null && !password.startsWith("$2a$10$")){
            BCryptPasswordEncoder bCryptPasswordEncoder = new BCryptPasswordEncoder();
            password = bCryptPasswordEncoder.encode(password);
            user.setPassword(password);
        }
        //更新用户表
        userDao.edit(user);
        //删除中间表
        userDao.deleteRoleIdsByUserId(user.getId());
        //向中间表中增加数据
        if(roleIds!=null && roleIds.length>0){
            for (Integer roleId : roleIds) {
                Map<String,Integer> map = new HashMap<>();
                map.put("roleId",roleId);
                map.put("userId",user.getId());
                userDao.setUserAndRole(map);
            }
        }
    }

    @Override
    public void deleteById(Integer id) {
        userDao.deleteRoleIdsByUserId(id);
        userDao.delete(id);
    }
}
